#pragma once
#include <iostream>
#include <cstdlib>

#include <fcntl.h>
#include <signal.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>

using namespace std;

const char *root = "/";
const char *dev_null = "/dev/null";

void Daemon(bool ischdir, bool isclose)
{
    // 1. 忽略引起程序异常退出的信号
    signal(SIGCHLD, SIG_IGN);
    signal(SIGPIPE, SIG_IGN);

    // 2. 不让自己称为组长
    if (fork() > 0)
        exit(0);

    // 3. 设置自己成为新的对话，后面代码实则是子进程在运行
    setsid();

    // 4. 每个进程都有自己的CWD, 是否将当前进程的CWD 更改为 '/' 根目录
    if (ischdir)
        chdir(root);

    // 5. 变成守护进程，不需要和用户的输入输出，错误进行关联
    if(isclose)
    {
        close(0);
        close(1);
        close(2);
    }
    else
    {
        int fd = open(dev_null, O_RDWR);
        if(fd > 0)
        {
            dup2(fd, 0);
            dup2(fd, 1);
            dup2(fd, 2);

            close(fd);
        }
    }
}